import ascendspeed_ops
from ascendspeed_ops import flash_attention
